package com.Hsu.mq.mqserver.datacenter;

/**
 * @author Hsu琛君珩
 * @ClassName:DataBaseManager
 * @date 2024-02-13
 * @apiNote
 * @Version: v1.0
 */

import com.Hsu.mq.DemoApplication;
import com.Hsu.mq.mqserver.core.Binding;
import com.Hsu.mq.mqserver.core.Exchange;
import com.Hsu.mq.mqserver.core.ExchangeType;
import com.Hsu.mq.mqserver.core.MSGQueue;
import com.Hsu.mq.mqserver.mapper.MetaMapper;

import java.io.File;
import java.util.List;

/**
 * 通过这个类整合上述的数据库操作
 * 数据库管理器，负责消息队列系统中的数据库操作，包括初始化数据库、建表、增删改查
 */
public class DataBaseManager {
    //要做的是从 Spring 中拿到现成的对象，不用 @Autowired 是因为 DataBaseManager 不是在 Spring 注册的对象，因为我们打算手动管理
    private MetaMapper metaMapper;

    /**
     * 初始化数据库，包括建立数据库连接、创建所需的数据表和插入初始化数据
     * 针对数据库进行初始化，建库建表 + 插入默认数据，在程序启动时
     * 1. 如果数据库已经存在了，有表了，不做任何操作
     * 2. 如果数据库不存在，则创建库、表，构造默认数据
     * 是否存在，就看 meta.db 文件是否存在即可
     */
    public void init(){
        //手动获取 MetaMapper
        metaMapper= DemoApplication.context.getBean(MetaMapper.class);
        //判断数据库是否已经存在
        if(!checkDBExists()){
            //数据库不存在就建表

            //先创建一个 data 目录
            File dataDir=new File("./data");
            dataDir.mkdirs();

            //创建数据表
            createTable();
            //插入默认数据
            createDefaultData();
            System.out.println("[DataBaseManager] 数据库初始化完成！");
        }else {
            //数据库已经存在就啥都不做
            System.out.println("[DataBaseManager] 数据库已经存在！");
        }
    }

    /**
     * 删除 meta.db 文件
     */
    public void deleteDB(){
        File file = new File("./data/meta.db");
        boolean ret = file.delete();
        if (ret) {
            System.out.println("[DataBaseManager] 删除数据库文件成功!");
        } else {
            System.out.println("[DataBaseManager] 删除数据库文件失败!");
        }
        File dataDir = new File("./data");
        ret=dataDir.delete();//delete() 删除目录要保证目录是空的
        if(ret){
            System.out.println("[DataBaseManager] 删除数据库目录成功！");
        }else {
            System.out.println("[DataBaseManager] 删除数据库目录失败！");
        }
    }

    /**
     * 创建默认数据，例如默认的交换机
     * 给数据库表中添加默认数据，主要是添加一个默认的 交换机，这是根据 RabbitMQ：带有一个 匿名 的交换机，类型是 DIRECT
     */
    private void createDefaultData() {
        //构造默认的交换机
        Exchange exchange = new Exchange();
        exchange.setName("");
        exchange.setType(ExchangeType.DIRECT);
        exchange.setDurable(true);
        exchange.setAutoDelete(false);
        metaMapper.insertExchange(exchange);
        System.out.println("[DataBaseManager] 创建初始数据完成!");
    }

    /**
     * 这个是建表的，不需要手动建库 (即不用创建 meta.db 文件)
     * 首次执行这里的数据库操作时 MyBatis 会自动创建 meta.db 的
     */
    private void createTable() {
        metaMapper.createExchangeTable();
        metaMapper.createQueueTable();
        metaMapper.createBindingTable();
        System.out.println("[DataBaseManager] 创建表完成！");
    }

    private boolean checkDBExists() {
        //url: jdbc:sqlite:./data/meta.db
        File file = new File("./data/meta.db");
        if (file.exists()) {
            return true;
        }
        return false;
    }

    //把其他数据库的操作也在这个类封装一下
    public void insertExchange(Exchange exchange){
        metaMapper.insertExchange(exchange);
    }
    public List<Exchange> selectAllExchanges(){
        return metaMapper.selectAllExchanges();
    }
    public void deleteExchange(String exchangeName){
        metaMapper.deleteExchange(exchangeName);
    }
    public void insertQueue(MSGQueue queue){
        metaMapper.insertQueue(queue);
    }
    public List<MSGQueue> selectAllQueues(){
        return metaMapper.selectAllQueues();
    }
    public void deleteQueue(String queueName){
        metaMapper.deleteQueue(queueName);
    }
    public void insertBinding(Binding binding){
        metaMapper.insertBinding(binding);
    }
    public List<Binding> selectAllBindings(){
        return metaMapper.selectAllBindings();
    }
    public void deleteBinding(Binding binding){
        metaMapper.deleteBinding(binding);
    }
}